"""
File   : 02_chat_model.py
Desc   : 对话型语言模型
Date   : 2024/11/17
Author : Tianyu Chen
"""
from langchain_openai import ChatOpenAI
from langchain.schema import HumanMessage

chat_model = ChatOpenAI()

TEXT = "给生产杯子的公司取一个中文名字"
messages = [HumanMessage(content=TEXT)]
# messages = [("human", TEXT)]  # 也可以这样写


if __name__ == "__main__":
    print(chat_model.invoke(messages))  # content='华杯制造有限公司' ...
